from WSOL.model import build_detector
from WSOL.dataset import build_dataset, build_preprocess_op
from torch.utils.data.dataloader import DataLoader
from WSOL.api.train import train
import yaml
from WSOL.model.module import BACKBONE
import cv2
import torch


yaml_file = open('./configs/base_detector.yaml', 'r', encoding='utf-8')
cfg = yaml.load(yaml_file)
yaml_file.close()


train(cfg)
